# fine-tuning the model
vgg16.trainable = False
for layer in vgg16.layers[-4:]:
    layer.trainable = True
    
history, model = classification(num_train_samples = 4000, 
                                learning_rate = 0.00001)
plot(history)

